'''
* This is the projet for Brtc LlmOps Platform
* @Author Leon-liao <liaosiliang@alltman.com>
* @Description //TODO 
* @File: 3_study_my_frist_chain.py
* @Time: 2025/7/13
* @All Rights Reserve By Brtc
'''
import dotenv
from langchain_community.chat_models import ChatOpenAI
from langchain_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate

dotenv.load_dotenv()

# 1、构建提示词
prompt = ChatPromptTemplate.from_template("""
你是OpenAi研发的机器人， 请按照人类的要求来回答问题， 
<history>
human:我叫博小睿很高兴认识你，
AI:博小睿你好，很高兴认识你
</history>
人类的问题是:{query}
""")
# 2、创建model
llm = ChatOpenAI(model="gpt-3.5-turbo-16k")
# 3、创建 paraser
parser = StrOutputParser()
# 4、构建链接
chain =  prompt| llm | parser
# 5、调用链
print(chain.invoke({"query", "请问我叫什么名字"}))